//
// Created by ZhihuaLiu on 21-1-5.
//

#ifndef SRC_PY_DECODE_H
#define SRC_PY_DECODE_H
#include<iostream>
#include <pybind11/pybind11.h>
#include <pybind11/numpy.h>
#include <pybind11/stl.h>
namespace py = pybind11;


int decode_head_cpu(py::array_t<float> out_cls,
                py::array_t<float> out_box,
                py::array_t<float> out_dir,
                py::array_t<float> out_pos,
                py::array_t<float> anchors,
                py::array_t<float> scores,
                py::array_t<float> Bbox,
                py::array_t<float> type,
                py::array_t<float> ClassThreshold,
                float pos_threshold,
                int kMaxResultSize,
                int filled_size_,
                int ClassNumber,
                int kRadianNumber,
                int kInfoNumber);
int nms_cpu_all(py::array_t<float> scores,
            py::array_t<float> Bbox,
            py::array_t<bool> keep,
            float iou_threshold,
            int filled_size_,
            int kInfoNumber);
#endif //SRC_PY_DECODE_H
